package controllers

import (
	"gitee.com/small_fish_official_net/beego_frame/bash"
	"github.com/astaxie/beego"
	"github.com/astaxie/beego/orm"
)

type Api struct {
	beego.Controller
}

// JSON返回数据
//
// 自动获取c.Data["json"]中的对象进行返回
func (c *Api) return_json(msg ...string) bool {
	if c.Data["json"] == nil {
		c.Data["json"] = map[string]string{}
	}
	if len(msg) == 0 {
		c.Data["json"] = map[string]interface{}{
			"code":       0,
			"message":    "成功",
			"serve_time": bash.Date(),
			"result":     c.Data["json"],
		}
	} else if len(msg) == 1 {
		c.Data["json"] = map[string]interface{}{
			"code":       1,
			"message":    msg[0],
			"serve_time": bash.Date(),
			"result":     map[string]string{},
		}
	}
	c.ServeJSON()
	return true
}

// 读操作的数据库
func (c *Api) get_orm() orm.Ormer {
	return bash.UseType(orm.NewOrm())
}

// 写操作的数据库
func (c *Api) set_orm() orm.Ormer {
	return bash.UseType(orm.NewOrm(), "write")
}
